# -*- coding: utf-8 -*-
# TencentBlueKing is pleased to support the open source community by making
# 蓝鲸智云 - PaaS 平台 (BlueKing - PaaS System) available.
# Copyright (C) 2017 THL A29 Limited, a Tencent company. All rights reserved.
# Licensed under the MIT License (the "License"); you may not use this file except
# in compliance with the License. You may obtain a copy of the License at
#
#     http://opensource.org/licenses/MIT
#
# Unless required by applicable law or agreed to in writing, software distributed under
# the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
# either express or implied. See the License for the specific language governing permissions and
# limitations under the License.
#
# We undertake not to change the open source license (MIT license) applicable
# to the current version of the project delivered to anyone in the future.

from rest_framework.pagination import LimitOffsetPagination, _positive_int


class MaxLimitOffsetPagination(LimitOffsetPagination):
    """限制最大分页数"""

    # 最多只能取 1000 条数据
    max_limit = 100
    max_offset = 900

    def get_offset(self, request):
        """LimitOffsetPagination 的 get_limit() 中处理了 max_limit"""
        try:
            return _positive_int(request.query_params[self.offset_query_param], strict=True, cutoff=self.max_offset)
        except (KeyError, ValueError):
            return 0
